import torch

class Config(object):
    num_layers = 5  # 层数
    hidden_dim = 300 # 神经元数目
    lr = 1e-3 # 学习率
    epoch = 500 # 学习轮数
    batch_size = 64 # batch大小
    word2vec_path = "model/WordEmbedding.txt"  # 词向量路径
    save_path = "model/NPM.model" # 模型保存路径
    train_wordpairs_path = "data/trainset_word" #训练词对集路径
    train_wordpairs_path = "data/train_indict_small"  # 开发词对集路径
    test_wordpairs_path = "data/test_indict_small"  # 训练词对集路径
    DEVICE = torch.device("cuda" if torch.cuda.is_available() else "cpu") #设备